/*
 * Sonatype Nexus (TM) Open Source Version
 * Copyright (c) 2008-present Sonatype, Inc.
 * All rights reserved. Includes the third-party code listed at http://links.sonatype.com/products/nexus/oss/attributions.
 *
 * This program and the accompanying materials are made available under the terms of the Eclipse Public License Version 1.0,
 * which accompanies this distribution and is available at http://www.eclipse.org/legal/epl-v10.html.
 *
 * Sonatype Nexus (TM) Professional Version is available from Sonatype, Inc. "Sonatype" and "Sonatype Nexus" are trademarks
 * of Sonatype, Inc. Apache Maven is a trademark of the Apache Software Foundation. M2eclipse is a trademark of the
 * Eclipse Foundation. All other trademarks are the property of their respective owners.
 */
package org.sonatype.nexus.repository.firewall.event;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.annotation.Nullable;

import org.sonatype.nexus.repository.vulnerability.AuditRepositoryComponent;

/**
 * @since 3.35
 */
public class QuarantinedComponentVersions
{
  private final Map<AuditRepositoryComponent, List<String>> auditRepositoryComponents = new HashMap<>();

  public void addComponentVersions(final QuarantinedComponentVersions componentVersions) {
    auditRepositoryComponents.putAll(componentVersions.getAuditRepositoryComponents());
  }

  /**
   * Add vulnerability to the component.
   *
   * @param auditRepositoryComponent {@link AuditRepositoryComponent} quarantined component.
   * @param version of the quarantined component
   */
  public void addRepositoryVersion(
      final AuditRepositoryComponent auditRepositoryComponent,
      @Nullable final String version)
  {
    List<String> versions = auditRepositoryComponents.computeIfAbsent(auditRepositoryComponent, k -> new ArrayList<>());
    if (version != null) {
      versions.add(version);
    }
  }

  public Map<AuditRepositoryComponent, List<String>> getAuditRepositoryComponents() {
    return auditRepositoryComponents;
  }
}
